package com.example.utils;


import com.example.entity.DataInfo;
import com.example.entity.ListDTO;

import java.util.ArrayList;
import java.util.List;

public class DataFormatUtils {

    public static ListDTO formatTableData2Bar(List<DataInfo> nameValues) {
        List<String> names = new ArrayList<>();
        List<String> values = new ArrayList<>();
        for (DataInfo keyValue : nameValues) {
            names.add(keyValue.getTitle() + "");
            values.add(keyValue.getCount() + "");
        }
        return ListDTO.builder().names(names).values(values).build();
    }
}